
"""
236. 二叉树的最近公共祖先
思路：通过遍历二叉树的形式，在后序遍历的位置，此时可以拿到左右子树遍历的结果，如果返回值不为空，说明来自该子树，此时根节点就是公共节点
"""
from com.siyuan.TreeNode import TreeNode


class Solution:

    def lowestCommonAncestor(self, root: 'TreeNode', p: 'TreeNode', q: 'TreeNode') -> 'TreeNode':
        if not root:
            return None

        # 如果根节点等于p和q，则返回根节点
        if root == p or root == q:
            return root

        # 搜索左右节点
        left = self.lowestCommonAncestor(root.left, p, q)
        right = self.lowestCommonAncestor(root.right, p, q)

        # 左右节点均不为空，则返回根节点
        if left and right:
            return root

        return left if left else right